CUDA_VISIBLE_DEVICES=5 python src/sequence_embedding.py \
    --fasta data/uniprot_sprot.fasta \
    --model /home/user4/workspace/protein_encoder/data/esm2_36_3B_UR50D \
    --batch_size 32 \
    --chunk_num 5 \
    --chunk_id 4 \
    --out_dir data/swissprot_esm2_3B